#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @File  : dwspark调用.py
# @Author: HCY
# @Date  : 2025/5/15
# @Desc  : 讯飞角色模拟服务dwspark调用示例

from dwspark.config import Config
# 加载系统环境变量：SPARKAI_UID、SPARKAI_APP_ID、SPARKAI_API_KEY、SPARKAI_API_SECRET

config = Config()

# SDK引入模型
from dwspark.models import CharacterSimulator
# 日志
from loguru import logger

'''
角色模拟器
'''
# 创建角色模拟器实例
model = CharacterSimulator(config)

# 示例1：创建玩家
logger.info('----------创建玩家----------')
player_id = model.create_player("测试玩家", "测试玩家身份描述")
logger.info(f"创建玩家结果，玩家ID: {player_id}")

# 示例2：修改玩家信息
if player_id:
    logger.info('----------修改玩家信息----------')
    result = model.modify_player(player_id, "修改后的玩家名称", "修改后的玩家身份描述")
    logger.info(f"修改玩家信息结果: {'成功' if result else '失败'}")

# 示例3：创建人格
if player_id:
    logger.info('----------创建人格----------')
    agent_id = model.create_agent(
        player_id=player_id,
        agent_name="测试人格",
        agent_identity="测试人格身份",
        agent_hobby="阅读、旅行",
        agent_personality_desc="性格开朗，善于沟通，喜欢帮助他人。"
    )
    logger.info(f"创建人格结果，人格ID: {agent_id}")

logger.info('done.')